package com.itheima.youdianda.controller;

import com.itheima.youdianda.common.R;
import com.itheima.youdianda.entity.UpdateFile;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;

@RestController
@RequestMapping("/common")
public class CommonController {
    @Value("${ydd.path}")
    private String basePath;

    /**
     * 上传文件
     * @param file
     * @param type
     * @return
     * @throws IOException
     */
    @PostMapping("/upload")
    public R<UpdateFile> upload(MultipartFile file, String type, HttpServletRequest request) throws IOException {
        //设定时间格式
        Date date = new Date();
        SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyyMMdd");
        //原始文件名
        String originalFilename = file.getOriginalFilename();
        //截取右边图片文件类型
        String suffix = originalFilename.substring(originalFilename.lastIndexOf("."));
        //获取当前时间毫秒值，并组合成新文件名
        String fileName = date.getTime() + suffix;
        //获取存储路径
        String savePath = "/upload/" + type + "/" + simpleDateFormat.format(date.getTime());


        //封装Filepath对象
        UpdateFile filePath=new UpdateFile();
        filePath.setSavePath(savePath+"/"+fileName);  //设置文件存储路径加文件
        filePath.setName(file.getOriginalFilename()); //设置文件名
        filePath.setSize((int) file.getSize());
        //filePath.setMime("image/png");
        //log.info(basePath+fileName);

        //判断文件上传类型
        if (type.equals("images")) {
            filePath.setMime("image/" + suffix);
        }
        //创建一个目录对象
        File dir = new File(basePath +savePath);  //创建文件夹
        //创建一个测试目录对象
        //File dirtest = new File(path);
        //判断当前目录是否存在
        if (!dir.exists()) {
            //目录不存在，重新创建
            dir.mkdirs();
        }

        try {
            //将临时文件转存到指定位置
            file.transferTo(new File(basePath + savePath+"/"+fileName)); //将文件转存到指定文件夹

        } catch (IOException e) {
            e.printStackTrace();
        }
        return R.success(filePath);

    }

    /**
     * 下载文件
     * @param name
     * @param response
     */
    @GetMapping("/download")
    public void download(String name, HttpServletResponse response){
        //通过输入流读取文件内容
        try {
            FileInputStream is=new FileInputStream(new File(basePath+name));
            //输出流,通过输出流将文件写回浏览器,在浏览器展示图片了
            ServletOutputStream os = response.getOutputStream();

            response.setContentType("image/jpeg");
            int len;
            byte[] bytes=new byte[1024];
            while((len=is.read(bytes))!=-1){
                os.write(bytes,0,len);
                os.flush();
            }
            is.close();
            os.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}
